package ynu.edu.model.service;


import ynu.edu.model.entity.ModelConfig;

import java.util.List;

/**
 * 模型配置服务接口
 */
public interface ModelConfigService {

    /**
     * 根据模型名称获取模型配置
     * @param modelName 模型名称
     * @return 模型配置
     */
    ModelConfig getModelConfig(String modelName);

    /**
     * 获取默认模型配置
     * @return 默认模型配置
     */
    ModelConfig getDefaultModel();

    /**
     * 获取所有启用的模型配置
     * @return 启用的模型配置列表
     */
    List<ModelConfig> getEnabledModels();

    /**
     * 获取Text2Img模型配置
     * @return Text2Img模型配置列表
     */
    List<ModelConfig> getText2ImgModels();

    /**
     * 获取Img2Img模型配置
     * @return Img2Img模型配置列表
     */
    List<ModelConfig> getImg2ImgModels();

    /**
     * 检查模型是否存在
     * @param modelName 模型名称
     * @return 是否存在
     */
    boolean existsModel(String modelName);

    /**
     * 获取用于参数构建的模型配置
     * @param modelName 模型名称
     * @return 模型配置
     */
    ModelConfig getModelForParams(String modelName);
} 